package com.stylefeng.guns.api.util;

import lombok.extern.slf4j.Slf4j;

/**
 * @ClassName TokenBucket
 * @Description TODO
 * @Author zhaoluowei
 * @Date 2019/9/21  20:16
 * @Version 1.0
 */
//令牌桶 对业务的峰值有一定的容忍度
@Slf4j
public class TokenBucket {

    private static int bucketNums = 1000; //桶的容量
    private static int rate = 2;// 流入速度 1ms 一个请求。
    private static int nowTokens;// 当前令牌数
    private static long timestamp = getNowTime();// 时间


    private static long getNowTime() {
        return System.currentTimeMillis();
    }


    private int min(int tokens) {
        if (bucketNums > tokens) {
            return tokens;
        } else {
            return bucketNums;
        }
    }

    public boolean getToken() {
        //记录 来拿令牌的时间
        long newTime = getNowTime();
        //添加令牌 (判断该有多少个令牌 )
        nowTokens = nowTokens + ((int) ((newTime - timestamp) * rate));
        //添加以后的令牌数量 与桶的容量 那个小
        nowTokens = min(nowTokens);
        System.out.println("当前令牌数量 --> " + nowTokens);
        //修改拿令牌的时间
        timestamp = getNowTime();
        //判断令牌是否足够
        if (nowTokens < 1) {
            return false;
        } else {
            nowTokens -= 1;
            return true;
        }
    }

//    public static void main(String[] args) throws InterruptedException {
//        TokenBucket tokenBucket = new TokenBucket();
//        for (int i = 0; i < 1000; i++) {
//            if (i == 10) {
//                Thread.sleep(500);
//            }
//            System.out.println("第 " + i + " 次 请求结果 = " + tokenBucket.getToken());
//        }
//
//    }

}
